AI that generates Python code